blob: 74873bc1e2246eb30291046b6f23fd6815461e07 [file] [log] [blame]
drhcce7d172000-05-31 15:34:51 +00001/*
2** Copyright (c) 1999, 2000 D. Richard Hipp
3**
4** This program is free software; you can redistribute it and/or
5** modify it under the terms of the GNU General Public
6** License as published by the Free Software Foundation; either
7** version 2 of the License, or (at your option) any later version.
8**
9** This program is distributed in the hope that it will be useful,
10** but WITHOUT ANY WARRANTY; without even the implied warranty of
11** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12** General Public License for more details.
13**
14** You should have received a copy of the GNU General Public
15** License along with this library; if not, write to the
16** Free Software Foundation, Inc., 59 Temple Place - Suite 330,
17** Boston, MA 02111-1307, USA.
18**
19** Author contact information:
20** drh@hwaci.com
21** http://www.hwaci.com/drh/
22**
23*************************************************************************
drh1ccde152000-06-17 13:12:39 +000024** This file contains routines used for analyzing expressions and
25** for generating VDBE code that evaluates expressions.
drhcce7d172000-05-31 15:34:51 +000026**
drh1ccde152000-06-17 13:12:39 +000027** $Id: expr.c,v 1.17 2000/06/17 13:12:40 drh Exp $
drhcce7d172000-05-31 15:34:51 +000028*/
29#include "sqliteInt.h"
30
31/*
drhfef52082000-06-06 01:50:43 +000032** Walk an expression tree. Return 1 if the expression is constant
33** and 0 if it involves variables.
34*/
35static int isConstant(Expr *p){
36 switch( p->op ){
37 case TK_ID:
38 case TK_FIELD:
39 case TK_DOT:
40 return 0;
41 default: {
42 if( p->pLeft && !isConstant(p->pLeft) ) return 0;
43 if( p->pRight && !isConstant(p->pRight) ) return 0;
44 if( p->pList ){
45 int i;
46 for(i=0; i<p->pList->nExpr; i++){
47 if( !isConstant(p->pList->a[i].pExpr) ) return 0;
48 }
49 }
50 break;
51 }
52 }
53 return 1;
54}
55
56/*
drh4794b982000-06-06 13:54:14 +000057** Walk the expression tree and process operators of the form:
58**
59** expr IN (SELECT ...)
60**
61** These operators have to be processed before field names are
62** resolved because each such operator increments pParse->nTab
drh1ccde152000-06-17 13:12:39 +000063** to reserve cursor numbers for its own use. But pParse->nTab
drh4794b982000-06-06 13:54:14 +000064** needs to be constant once we begin resolving field names.
65**
66** Actually, the processing of IN-SELECT is only started by this
67** routine. This routine allocates a cursor number to the IN-SELECT
68** and then moves on. The code generation is done by
69** sqliteExprResolveIds() which must be called afterwards.
70*/
71void sqliteExprResolveInSelect(Parse *pParse, Expr *pExpr){
72 if( pExpr==0 ) return;
73 if( pExpr->op==TK_IN && pExpr->pSelect!=0 ){
74 pExpr->iTable = pParse->nTab++;
75 }else{
76 if( pExpr->pLeft ) sqliteExprResolveInSelect(pParse, pExpr->pLeft);
77 if( pExpr->pRight ) sqliteExprResolveInSelect(pParse, pExpr->pRight);
78 if( pExpr->pList ){
79 int i;
80 ExprList *pList = pExpr->pList;
81 for(i=0; i<pList->nExpr; i++){
82 sqliteExprResolveInSelect(pParse, pList->a[i].pExpr);
83 }
84 }
85 }
86}
87
88/*
drhcce7d172000-05-31 15:34:51 +000089** This routine walks an expression tree and resolves references to
90** table fields. Nodes of the form ID.ID or ID resolve into an
91** index to the table in the table list and a field offset. The opcode
92** for such nodes is changed to TK_FIELD. The iTable value is changed
drh19a775c2000-06-05 18:54:46 +000093** to the index of the referenced table in pTabList plus the pParse->nTab
94** value. The iField value is changed to the index of the field of the
95** referenced table.
96**
drhfef52082000-06-06 01:50:43 +000097** We also check for instances of the IN operator. IN comes in two
98** forms:
99**
100** expr IN (exprlist)
101** and
102** expr IN (SELECT ...)
103**
104** The first form is handled by creating a set holding the list
105** of allowed values. The second form causes the SELECT to generate
106** a temporary table.
107**
108** This routine also looks for scalar SELECTs that are part of an expression.
drh19a775c2000-06-05 18:54:46 +0000109** If it finds any, it generates code to write the value of that select
110** into a memory cell.
drhcce7d172000-05-31 15:34:51 +0000111**
112** Unknown fields or tables provoke an error. The function returns
113** the number of errors seen and leaves an error message on pParse->zErrMsg.
114*/
115int sqliteExprResolveIds(Parse *pParse, IdList *pTabList, Expr *pExpr){
116 if( pExpr==0 ) return 0;
117 switch( pExpr->op ){
118 /* A lone identifier */
119 case TK_ID: {
120 int cnt = 0; /* Number of matches */
121 int i; /* Loop counter */
drh6e142f52000-06-08 13:36:40 +0000122 char *z = sqliteStrNDup(pExpr->token.z, pExpr->token.n);
drhcce7d172000-05-31 15:34:51 +0000123 for(i=0; i<pTabList->nId; i++){
124 int j;
125 Table *pTab = pTabList->a[i].pTab;
126 if( pTab==0 ) continue;
127 for(j=0; j<pTab->nCol; j++){
drh7020f652000-06-03 18:06:52 +0000128 if( sqliteStrICmp(pTab->aCol[j].zName, z)==0 ){
drhcce7d172000-05-31 15:34:51 +0000129 cnt++;
drh19a775c2000-06-05 18:54:46 +0000130 pExpr->iTable = i + pParse->nTab;
drhcce7d172000-05-31 15:34:51 +0000131 pExpr->iField = j;
132 }
133 }
134 }
135 sqliteFree(z);
136 if( cnt==0 ){
137 sqliteSetNString(&pParse->zErrMsg, "no such field: ", -1,
138 pExpr->token.z, pExpr->token.n, 0);
139 pParse->nErr++;
140 return 1;
141 }else if( cnt>1 ){
142 sqliteSetNString(&pParse->zErrMsg, "ambiguous field name: ", -1,
143 pExpr->token.z, pExpr->token.n, 0);
144 pParse->nErr++;
145 return 1;
146 }
147 pExpr->op = TK_FIELD;
148 break;
149 }
150
151 /* A table name and field name: ID.ID */
152 case TK_DOT: {
drhbed86902000-06-02 13:27:59 +0000153 int cnt = 0; /* Number of matches */
154 int i; /* Loop counter */
drhcce7d172000-05-31 15:34:51 +0000155 Expr *pLeft, *pRight; /* Left and right subbranches of the expr */
drhcce7d172000-05-31 15:34:51 +0000156 char *zLeft, *zRight; /* Text of an identifier */
157
158 pLeft = pExpr->pLeft;
159 pRight = pExpr->pRight;
160 assert( pLeft && pLeft->op==TK_ID );
161 assert( pRight && pRight->op==TK_ID );
drh6e142f52000-06-08 13:36:40 +0000162 zLeft = sqliteStrNDup(pLeft->token.z, pLeft->token.n);
163 zRight = sqliteStrNDup(pRight->token.z, pRight->token.n);
drhcce7d172000-05-31 15:34:51 +0000164 for(i=0; i<pTabList->nId; i++){
165 int j;
166 char *zTab;
167 Table *pTab = pTabList->a[i].pTab;
168 if( pTab==0 ) continue;
169 if( pTabList->a[i].zAlias ){
170 zTab = pTabList->a[i].zAlias;
171 }else{
172 zTab = pTab->zName;
173 }
174 if( sqliteStrICmp(zTab, zLeft)!=0 ) continue;
175 for(j=0; j<pTab->nCol; j++){
drh7020f652000-06-03 18:06:52 +0000176 if( sqliteStrICmp(pTab->aCol[j].zName, zRight)==0 ){
drhcce7d172000-05-31 15:34:51 +0000177 cnt++;
drh19a775c2000-06-05 18:54:46 +0000178 pExpr->iTable = i + pParse->nTab;
drhcce7d172000-05-31 15:34:51 +0000179 pExpr->iField = j;
180 }
181 }
182 }
183 sqliteFree(zLeft);
184 sqliteFree(zRight);
185 if( cnt==0 ){
186 sqliteSetNString(&pParse->zErrMsg, "no such field: ", -1,
187 pLeft->token.z, pLeft->token.n, ".", 1,
188 pRight->token.z, pRight->token.n, 0);
189 pParse->nErr++;
190 return 1;
191 }else if( cnt>1 ){
192 sqliteSetNString(&pParse->zErrMsg, "ambiguous field name: ", -1,
193 pLeft->token.z, pLeft->token.n, ".", 1,
194 pRight->token.z, pRight->token.n, 0);
195 pParse->nErr++;
196 return 1;
197 }
198 sqliteExprDelete(pLeft);
199 pExpr->pLeft = 0;
200 sqliteExprDelete(pRight);
201 pExpr->pRight = 0;
202 pExpr->op = TK_FIELD;
203 break;
204 }
205
drhfef52082000-06-06 01:50:43 +0000206 case TK_IN: {
drhd8bc7082000-06-07 23:51:50 +0000207 Vdbe *v = sqliteGetVdbe(pParse);
drhfef52082000-06-06 01:50:43 +0000208 if( v==0 ) return 1;
drhcfab11b2000-06-06 03:31:22 +0000209 if( sqliteExprResolveIds(pParse, pTabList, pExpr->pLeft) ){
210 return 1;
211 }
drhfef52082000-06-06 01:50:43 +0000212 if( pExpr->pSelect ){
213 /* Case 1: expr IN (SELECT ...)
214 **
215 ** Generate code to write the results of the select into a temporary
drh4794b982000-06-06 13:54:14 +0000216 ** table. The cursor number of the temporary table has already
217 ** been put in iTable by sqliteExprResolveInSelect().
drhfef52082000-06-06 01:50:43 +0000218 */
drh4794b982000-06-06 13:54:14 +0000219 sqliteVdbeAddOp(v, OP_Open, pExpr->iTable, 1, 0, 0);
drhfef52082000-06-06 01:50:43 +0000220 if( sqliteSelect(pParse, pExpr->pSelect, SRT_Set, pExpr->iTable) );
221 }else if( pExpr->pList ){
222 /* Case 2: expr IN (exprlist)
223 **
224 ** Create a set to put the exprlist values in. The Set id is stored
225 ** in iTable.
226 */
227 int i, iSet;
228 for(i=0; i<pExpr->pList->nExpr; i++){
229 Expr *pE2 = pExpr->pList->a[i].pExpr;
drhfef52082000-06-06 01:50:43 +0000230 if( !isConstant(pE2) ){
231 sqliteSetString(&pParse->zErrMsg,
232 "right-hand side of IN operator must be constant", 0);
233 pParse->nErr++;
234 return 1;
235 }
drh4794b982000-06-06 13:54:14 +0000236 if( sqliteExprCheck(pParse, pE2, 0, 0) ){
237 return 1;
238 }
drhfef52082000-06-06 01:50:43 +0000239 }
240 iSet = pExpr->iTable = pParse->nSet++;
241 for(i=0; i<pExpr->pList->nExpr; i++){
242 Expr *pE2 = pExpr->pList->a[i].pExpr;
243 switch( pE2->op ){
244 case TK_FLOAT:
245 case TK_INTEGER:
246 case TK_STRING: {
247 int addr = sqliteVdbeAddOp(v, OP_SetInsert, iSet, 0, 0, 0);
248 sqliteVdbeChangeP3(v, addr, pE2->token.z, pE2->token.n);
249 sqliteVdbeDequoteP3(v, addr);
250 break;
251 }
252 default: {
253 sqliteExprCode(pParse, pE2);
254 sqliteVdbeAddOp(v, OP_SetInsert, iSet, 0, 0, 0);
255 break;
256 }
257 }
258 }
259 }
drhcfab11b2000-06-06 03:31:22 +0000260 break;
drhfef52082000-06-06 01:50:43 +0000261 }
262
drh19a775c2000-06-05 18:54:46 +0000263 case TK_SELECT: {
drhfef52082000-06-06 01:50:43 +0000264 /* This has to be a scalar SELECT. Generate code to put the
265 ** value of this select in a memory cell and record the number
266 ** of the memory cell in iField.
267 */
drh19a775c2000-06-05 18:54:46 +0000268 pExpr->iField = pParse->nMem++;
drhfef52082000-06-06 01:50:43 +0000269 if( sqliteSelect(pParse, pExpr->pSelect, SRT_Mem, pExpr->iField) ){
drh19a775c2000-06-05 18:54:46 +0000270 return 1;
271 }
272 break;
273 }
274
drhcce7d172000-05-31 15:34:51 +0000275 /* For all else, just recursively walk the tree */
276 default: {
drh4794b982000-06-06 13:54:14 +0000277 if( pExpr->pLeft
278 && sqliteExprResolveIds(pParse, pTabList, pExpr->pLeft) ){
drhcce7d172000-05-31 15:34:51 +0000279 return 1;
280 }
281 if( pExpr->pRight
drh4794b982000-06-06 13:54:14 +0000282 && sqliteExprResolveIds(pParse, pTabList, pExpr->pRight) ){
drhcce7d172000-05-31 15:34:51 +0000283 return 1;
284 }
285 if( pExpr->pList ){
286 int i;
287 ExprList *pList = pExpr->pList;
288 for(i=0; i<pList->nExpr; i++){
289 if( sqliteExprResolveIds(pParse, pTabList, pList->a[i].pExpr) ){
290 return 1;
291 }
292 }
293 }
294 }
295 }
296 return 0;
297}
298
299#if 0 /* NOT USED */
300/*
301** Compare a token against a string. Return TRUE if they match.
302*/
303static int sqliteTokenCmp(Token *pToken, const char *zStr){
304 int n = strlen(zStr);
305 if( n!=pToken->n ) return 0;
306 return sqliteStrNICmp(pToken->z, zStr, n)==0;
307}
308#endif
309
310/*
311** Convert a function name into its integer identifier. Return the
312** identifier. Return FN_Unknown if the function name is unknown.
313*/
314int sqliteFuncId(Token *pToken){
315 static const struct {
316 char *zName;
317 int len;
318 int id;
319 } aFunc[] = {
320 { "count", 5, FN_Count },
321 { "min", 3, FN_Min },
322 { "max", 3, FN_Max },
323 { "sum", 3, FN_Sum },
drh22827922000-06-06 17:27:05 +0000324 { "avg", 3, FN_Avg },
drh0bdaf622000-06-11 23:50:13 +0000325 { "fcnt", 4, FN_Fcnt }, /* Used for testing only */
drhcce7d172000-05-31 15:34:51 +0000326 };
327 int i;
328 for(i=0; i<ArraySize(aFunc); i++){
329 if( aFunc[i].len==pToken->n
330 && sqliteStrNICmp(pToken->z, aFunc[i].zName, aFunc[i].len)==0 ){
331 return aFunc[i].id;
332 }
333 }
334 return FN_Unknown;
335}
336
337/*
338** Error check the functions in an expression. Make sure all
339** function names are recognized and all functions have the correct
340** number of arguments. Leave an error message in pParse->zErrMsg
341** if anything is amiss. Return the number of errors.
342**
343** if pIsAgg is not null and this expression is an aggregate function
344** (like count(*) or max(value)) then write a 1 into *pIsAgg.
345*/
346int sqliteExprCheck(Parse *pParse, Expr *pExpr, int allowAgg, int *pIsAgg){
347 int nErr = 0;
348 if( pExpr==0 ) return 0;
drhcce7d172000-05-31 15:34:51 +0000349 switch( pExpr->op ){
350 case TK_FUNCTION: {
351 int id = sqliteFuncId(&pExpr->token);
352 int n = pExpr->pList ? pExpr->pList->nExpr : 0;
353 int no_such_func = 0;
354 int too_many_args = 0;
355 int too_few_args = 0;
356 int is_agg = 0;
357 int i;
drh22827922000-06-06 17:27:05 +0000358 pExpr->iField = id;
drhcce7d172000-05-31 15:34:51 +0000359 switch( id ){
360 case FN_Unknown: {
361 no_such_func = 1;
362 break;
363 }
364 case FN_Count: {
365 no_such_func = !allowAgg;
366 too_many_args = n>1;
367 is_agg = 1;
368 break;
369 }
370 case FN_Max:
371 case FN_Min: {
372 too_few_args = allowAgg ? n<1 : n<2;
373 is_agg = n==1;
374 break;
375 }
drh22827922000-06-06 17:27:05 +0000376 case FN_Avg:
drhcce7d172000-05-31 15:34:51 +0000377 case FN_Sum: {
378 no_such_func = !allowAgg;
379 too_many_args = n>1;
380 too_few_args = n<1;
381 is_agg = 1;
382 break;
383 }
drh0bdaf622000-06-11 23:50:13 +0000384 /* The "fcnt(*)" function always returns the number of fetch
385 ** operations that have occurred so far while processing the
386 ** SQL statement. This information can be used by test procedures
387 ** to verify that indices are being used properly to minimize
388 ** searching. All arguments to fcnt() are ignored. fcnt() has
389 ** no use (other than testing) that we are aware of.
390 */
391 case FN_Fcnt: {
392 n = 0;
393 break;
394 }
drhcce7d172000-05-31 15:34:51 +0000395 default: break;
396 }
397 if( no_such_func ){
398 sqliteSetNString(&pParse->zErrMsg, "no such function: ", -1,
399 pExpr->token.z, pExpr->token.n, 0);
400 pParse->nErr++;
401 nErr++;
402 }else if( too_many_args ){
403 sqliteSetNString(&pParse->zErrMsg, "too many arguments to function ",-1,
404 pExpr->token.z, pExpr->token.n, "()", 2, 0);
405 pParse->nErr++;
406 nErr++;
407 }else if( too_few_args ){
408 sqliteSetNString(&pParse->zErrMsg, "too few arguments to function ",-1,
409 pExpr->token.z, pExpr->token.n, "()", 2, 0);
410 pParse->nErr++;
411 nErr++;
412 }
drh22827922000-06-06 17:27:05 +0000413 if( is_agg ) pExpr->op = TK_AGG_FUNCTION;
drhcce7d172000-05-31 15:34:51 +0000414 if( is_agg && pIsAgg ) *pIsAgg = 1;
415 for(i=0; nErr==0 && i<n; i++){
drh4cfa7932000-06-08 15:10:46 +0000416 nErr = sqliteExprCheck(pParse, pExpr->pList->a[i].pExpr,
417 allowAgg && !is_agg, pIsAgg);
drhcce7d172000-05-31 15:34:51 +0000418 }
419 }
420 default: {
421 if( pExpr->pLeft ){
drh22827922000-06-06 17:27:05 +0000422 nErr = sqliteExprCheck(pParse, pExpr->pLeft, allowAgg, pIsAgg);
drhcce7d172000-05-31 15:34:51 +0000423 }
424 if( nErr==0 && pExpr->pRight ){
drh22827922000-06-06 17:27:05 +0000425 nErr = sqliteExprCheck(pParse, pExpr->pRight, allowAgg, pIsAgg);
drhcce7d172000-05-31 15:34:51 +0000426 }
drhfef52082000-06-06 01:50:43 +0000427 if( nErr==0 && pExpr->pList ){
428 int n = pExpr->pList->nExpr;
429 int i;
430 for(i=0; nErr==0 && i<n; i++){
drh22827922000-06-06 17:27:05 +0000431 Expr *pE2 = pExpr->pList->a[i].pExpr;
432 nErr = sqliteExprCheck(pParse, pE2, allowAgg, pIsAgg);
drhfef52082000-06-06 01:50:43 +0000433 }
434 }
drhcce7d172000-05-31 15:34:51 +0000435 break;
436 }
437 }
438 return nErr;
439}
440
441/*
442** Generate code into the current Vdbe to evaluate the given
drh1ccde152000-06-17 13:12:39 +0000443** expression and leave the result on the top of stack.
drhcce7d172000-05-31 15:34:51 +0000444*/
445void sqliteExprCode(Parse *pParse, Expr *pExpr){
446 Vdbe *v = pParse->pVdbe;
447 int op;
448 switch( pExpr->op ){
449 case TK_PLUS: op = OP_Add; break;
450 case TK_MINUS: op = OP_Subtract; break;
451 case TK_STAR: op = OP_Multiply; break;
452 case TK_SLASH: op = OP_Divide; break;
453 case TK_AND: op = OP_And; break;
454 case TK_OR: op = OP_Or; break;
455 case TK_LT: op = OP_Lt; break;
456 case TK_LE: op = OP_Le; break;
457 case TK_GT: op = OP_Gt; break;
458 case TK_GE: op = OP_Ge; break;
459 case TK_NE: op = OP_Ne; break;
460 case TK_EQ: op = OP_Eq; break;
461 case TK_LIKE: op = OP_Like; break;
462 case TK_GLOB: op = OP_Glob; break;
463 case TK_ISNULL: op = OP_IsNull; break;
464 case TK_NOTNULL: op = OP_NotNull; break;
465 case TK_NOT: op = OP_Not; break;
466 case TK_UMINUS: op = OP_Negative; break;
467 default: break;
468 }
469 switch( pExpr->op ){
470 case TK_FIELD: {
drh22827922000-06-06 17:27:05 +0000471 if( pParse->useAgg ){
472 sqliteVdbeAddOp(v, OP_AggGet, 0, pExpr->iAgg, 0, 0);
473 }else{
474 sqliteVdbeAddOp(v, OP_Field, pExpr->iTable, pExpr->iField, 0, 0);
475 }
drhcce7d172000-05-31 15:34:51 +0000476 break;
477 }
478 case TK_INTEGER: {
479 int i = atoi(pExpr->token.z);
480 sqliteVdbeAddOp(v, OP_Integer, i, 0, 0, 0);
481 break;
482 }
483 case TK_FLOAT: {
484 int addr = sqliteVdbeAddOp(v, OP_String, 0, 0, 0, 0);
485 sqliteVdbeChangeP3(v, addr, pExpr->token.z, pExpr->token.n);
486 break;
487 }
488 case TK_STRING: {
489 int addr = sqliteVdbeAddOp(v, OP_String, 0, 0, 0, 0);
490 sqliteVdbeChangeP3(v, addr, pExpr->token.z, pExpr->token.n);
491 sqliteVdbeDequoteP3(v, addr);
492 break;
493 }
494 case TK_NULL: {
drhc61053b2000-06-04 12:58:36 +0000495 sqliteVdbeAddOp(v, OP_Null, 0, 0, 0, 0);
drhcce7d172000-05-31 15:34:51 +0000496 break;
497 }
498 case TK_AND:
499 case TK_OR:
500 case TK_PLUS:
501 case TK_STAR:
502 case TK_MINUS:
503 case TK_SLASH: {
504 sqliteExprCode(pParse, pExpr->pLeft);
505 sqliteExprCode(pParse, pExpr->pRight);
506 sqliteVdbeAddOp(v, op, 0, 0, 0, 0);
507 break;
508 }
drh00400772000-06-16 20:51:26 +0000509 case TK_CONCAT: {
510 sqliteExprCode(pParse, pExpr->pLeft);
511 sqliteExprCode(pParse, pExpr->pRight);
512 sqliteVdbeAddOp(v, OP_Concat, 2, 0, 0, 0);
513 break;
514 }
drhcce7d172000-05-31 15:34:51 +0000515 case TK_LT:
516 case TK_LE:
517 case TK_GT:
518 case TK_GE:
519 case TK_NE:
520 case TK_EQ:
521 case TK_LIKE:
522 case TK_GLOB: {
523 int dest;
524 sqliteVdbeAddOp(v, OP_Integer, 1, 0, 0, 0);
525 sqliteExprCode(pParse, pExpr->pLeft);
526 sqliteExprCode(pParse, pExpr->pRight);
527 dest = sqliteVdbeCurrentAddr(v) + 2;
528 sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
529 sqliteVdbeAddOp(v, OP_AddImm, -1, 0, 0, 0);
530 break;
531 }
drhcce7d172000-05-31 15:34:51 +0000532 case TK_UMINUS: {
drh6e142f52000-06-08 13:36:40 +0000533 assert( pExpr->pLeft );
534 if( pExpr->pLeft->op==TK_INTEGER ){
535 int i = atoi(pExpr->pLeft->token.z);
536 sqliteVdbeAddOp(v, OP_Integer, -i, 0, 0, 0);
537 break;
538 }else if( pExpr->pLeft->op==TK_FLOAT ){
539 Token *p = &pExpr->pLeft->token;
540 char *z = sqliteMalloc( p->n + 2 );
541 sprintf(z, "-%.*s", p->n, p->z);
542 sqliteVdbeAddOp(v, OP_String, 0, 0, z, 0);
543 sqliteFree(z);
544 break;
545 }
drh1ccde152000-06-17 13:12:39 +0000546 /* Fall through into TK_NOT */
drh6e142f52000-06-08 13:36:40 +0000547 }
548 case TK_NOT: {
drhcce7d172000-05-31 15:34:51 +0000549 sqliteExprCode(pParse, pExpr->pLeft);
550 sqliteVdbeAddOp(v, op, 0, 0, 0, 0);
551 break;
552 }
553 case TK_ISNULL:
554 case TK_NOTNULL: {
555 int dest;
drh8be51132000-06-03 19:19:41 +0000556 sqliteVdbeAddOp(v, OP_Integer, 1, 0, 0, 0);
drhcce7d172000-05-31 15:34:51 +0000557 sqliteExprCode(pParse, pExpr->pLeft);
558 dest = sqliteVdbeCurrentAddr(v) + 2;
559 sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
drh8be51132000-06-03 19:19:41 +0000560 sqliteVdbeAddOp(v, OP_AddImm, -1, 0, 0, 0);
drhcce7d172000-05-31 15:34:51 +0000561 break;
562 }
drh22827922000-06-06 17:27:05 +0000563 case TK_AGG_FUNCTION: {
564 sqliteVdbeAddOp(v, OP_AggGet, 0, pExpr->iAgg, 0, 0);
565 if( pExpr->iField==FN_Avg ){
566 assert( pParse->iAggCount>=0 && pParse->iAggCount<pParse->nAgg );
567 sqliteVdbeAddOp(v, OP_AggGet, 0, pParse->iAggCount, 0, 0);
568 sqliteVdbeAddOp(v, OP_Divide, 0, 0, 0, 0);
569 }
570 break;
571 }
drhcce7d172000-05-31 15:34:51 +0000572 case TK_FUNCTION: {
drh22827922000-06-06 17:27:05 +0000573 int id = pExpr->iField;
drhcce7d172000-05-31 15:34:51 +0000574 int op;
575 int i;
576 ExprList *pList = pExpr->pList;
drh0bdaf622000-06-11 23:50:13 +0000577 if( id==FN_Fcnt ){
578 sqliteVdbeAddOp(v, OP_Fcnt, 0, 0, 0, 0);
579 break;
580 }
drhcce7d172000-05-31 15:34:51 +0000581 op = id==FN_Min ? OP_Min : OP_Max;
582 for(i=0; i<pList->nExpr; i++){
583 sqliteExprCode(pParse, pList->a[i].pExpr);
584 if( i>0 ){
585 sqliteVdbeAddOp(v, op, 0, 0, 0, 0);
586 }
587 }
588 break;
589 }
drh19a775c2000-06-05 18:54:46 +0000590 case TK_SELECT: {
591 sqliteVdbeAddOp(v, OP_MemLoad, pExpr->iField, 0, 0, 0);
592 break;
593 }
drhfef52082000-06-06 01:50:43 +0000594 case TK_IN: {
595 int addr;
drh4794b982000-06-06 13:54:14 +0000596 sqliteVdbeAddOp(v, OP_Integer, 1, 0, 0, 0);
drhfef52082000-06-06 01:50:43 +0000597 sqliteExprCode(pParse, pExpr->pLeft);
598 addr = sqliteVdbeCurrentAddr(v);
599 if( pExpr->pSelect ){
600 sqliteVdbeAddOp(v, OP_Found, pExpr->iTable, addr+2, 0, 0);
601 }else{
602 sqliteVdbeAddOp(v, OP_SetFound, pExpr->iTable, addr+2, 0, 0);
603 }
drh4794b982000-06-06 13:54:14 +0000604 sqliteVdbeAddOp(v, OP_AddImm, -1, 0, 0, 0);
drhfef52082000-06-06 01:50:43 +0000605 break;
606 }
607 case TK_BETWEEN: {
608 int lbl = sqliteVdbeMakeLabel(v);
609 sqliteVdbeAddOp(v, OP_Integer, 0, 0, 0, 0);
610 sqliteExprIfFalse(pParse, pExpr, lbl);
611 sqliteVdbeAddOp(v, OP_AddImm, 1, 0, 0, 0);
612 sqliteVdbeResolveLabel(v, lbl);
613 break;
614 }
drhcce7d172000-05-31 15:34:51 +0000615 }
616 return;
617}
618
619/*
620** Generate code for a boolean expression such that a jump is made
621** to the label "dest" if the expression is true but execution
622** continues straight thru if the expression is false.
623*/
624void sqliteExprIfTrue(Parse *pParse, Expr *pExpr, int dest){
625 Vdbe *v = pParse->pVdbe;
626 int op = 0;
627 switch( pExpr->op ){
628 case TK_LT: op = OP_Lt; break;
629 case TK_LE: op = OP_Le; break;
630 case TK_GT: op = OP_Gt; break;
631 case TK_GE: op = OP_Ge; break;
632 case TK_NE: op = OP_Ne; break;
633 case TK_EQ: op = OP_Eq; break;
634 case TK_LIKE: op = OP_Like; break;
635 case TK_GLOB: op = OP_Glob; break;
636 case TK_ISNULL: op = OP_IsNull; break;
637 case TK_NOTNULL: op = OP_NotNull; break;
638 default: break;
639 }
640 switch( pExpr->op ){
641 case TK_AND: {
642 int d2 = sqliteVdbeMakeLabel(v);
643 sqliteExprIfFalse(pParse, pExpr->pLeft, d2);
644 sqliteExprIfTrue(pParse, pExpr->pRight, dest);
645 sqliteVdbeResolveLabel(v, d2);
646 break;
647 }
648 case TK_OR: {
649 sqliteExprIfTrue(pParse, pExpr->pLeft, dest);
650 sqliteExprIfTrue(pParse, pExpr->pRight, dest);
651 break;
652 }
653 case TK_NOT: {
654 sqliteExprIfFalse(pParse, pExpr->pLeft, dest);
655 break;
656 }
657 case TK_LT:
658 case TK_LE:
659 case TK_GT:
660 case TK_GE:
661 case TK_NE:
662 case TK_EQ:
663 case TK_LIKE:
664 case TK_GLOB: {
665 sqliteExprCode(pParse, pExpr->pLeft);
666 sqliteExprCode(pParse, pExpr->pRight);
667 sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
668 break;
669 }
670 case TK_ISNULL:
671 case TK_NOTNULL: {
672 sqliteExprCode(pParse, pExpr->pLeft);
673 sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
674 break;
675 }
drhfef52082000-06-06 01:50:43 +0000676 case TK_IN: {
drhcfab11b2000-06-06 03:31:22 +0000677 sqliteExprCode(pParse, pExpr->pLeft);
drhfef52082000-06-06 01:50:43 +0000678 if( pExpr->pSelect ){
679 sqliteVdbeAddOp(v, OP_Found, pExpr->iTable, dest, 0, 0);
680 }else{
681 sqliteVdbeAddOp(v, OP_SetFound, pExpr->iTable, dest, 0, 0);
682 }
683 break;
684 }
685 case TK_BETWEEN: {
686 int lbl = sqliteVdbeMakeLabel(v);
687 sqliteExprCode(pParse, pExpr->pLeft);
688 sqliteVdbeAddOp(v, OP_Dup, 0, 0, 0, 0);
689 sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
690 sqliteVdbeAddOp(v, OP_Lt, 0, lbl, 0, 0);
691 sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
692 sqliteVdbeAddOp(v, OP_Le, 0, dest, 0, 0);
693 sqliteVdbeAddOp(v, OP_Integer, 0, 0, 0, 0);
694 sqliteVdbeAddOp(v, OP_Pop, 1, 0, 0, lbl);
695 break;
696 }
drhcce7d172000-05-31 15:34:51 +0000697 default: {
698 sqliteExprCode(pParse, pExpr);
699 sqliteVdbeAddOp(v, OP_If, 0, dest, 0, 0);
700 break;
701 }
702 }
703}
704
705/*
706** Generate code for boolean expression such that a jump is made
707** to the label "dest" if the expression is false but execution
708** continues straight thru if the expression is true.
709*/
710void sqliteExprIfFalse(Parse *pParse, Expr *pExpr, int dest){
711 Vdbe *v = pParse->pVdbe;
712 int op = 0;
713 switch( pExpr->op ){
714 case TK_LT: op = OP_Ge; break;
715 case TK_LE: op = OP_Gt; break;
716 case TK_GT: op = OP_Le; break;
717 case TK_GE: op = OP_Lt; break;
718 case TK_NE: op = OP_Eq; break;
719 case TK_EQ: op = OP_Ne; break;
720 case TK_LIKE: op = OP_Like; break;
721 case TK_GLOB: op = OP_Glob; break;
722 case TK_ISNULL: op = OP_NotNull; break;
723 case TK_NOTNULL: op = OP_IsNull; break;
724 default: break;
725 }
726 switch( pExpr->op ){
727 case TK_AND: {
728 sqliteExprIfFalse(pParse, pExpr->pLeft, dest);
729 sqliteExprIfFalse(pParse, pExpr->pRight, dest);
730 break;
731 }
732 case TK_OR: {
733 int d2 = sqliteVdbeMakeLabel(v);
734 sqliteExprIfTrue(pParse, pExpr->pLeft, d2);
735 sqliteExprIfFalse(pParse, pExpr->pRight, dest);
736 sqliteVdbeResolveLabel(v, d2);
737 break;
738 }
739 case TK_NOT: {
740 sqliteExprIfTrue(pParse, pExpr->pLeft, dest);
741 break;
742 }
743 case TK_LT:
744 case TK_LE:
745 case TK_GT:
746 case TK_GE:
747 case TK_NE:
748 case TK_EQ: {
749 sqliteExprCode(pParse, pExpr->pLeft);
750 sqliteExprCode(pParse, pExpr->pRight);
751 sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
752 break;
753 }
754 case TK_LIKE:
755 case TK_GLOB: {
756 sqliteExprCode(pParse, pExpr->pLeft);
757 sqliteExprCode(pParse, pExpr->pRight);
758 sqliteVdbeAddOp(v, op, 1, dest, 0, 0);
759 break;
760 }
761 case TK_ISNULL:
762 case TK_NOTNULL: {
763 sqliteExprCode(pParse, pExpr->pLeft);
764 sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
765 break;
766 }
drhfef52082000-06-06 01:50:43 +0000767 case TK_IN: {
drhcfab11b2000-06-06 03:31:22 +0000768 sqliteExprCode(pParse, pExpr->pLeft);
drhfef52082000-06-06 01:50:43 +0000769 if( pExpr->pSelect ){
770 sqliteVdbeAddOp(v, OP_NotFound, pExpr->iTable, dest, 0, 0);
771 }else{
772 sqliteVdbeAddOp(v, OP_SetNotFound, pExpr->iTable, dest, 0, 0);
773 }
774 break;
775 }
776 case TK_BETWEEN: {
777 int addr;
778 sqliteExprCode(pParse, pExpr->pLeft);
779 sqliteVdbeAddOp(v, OP_Dup, 0, 0, 0, 0);
780 sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
781 addr = sqliteVdbeCurrentAddr(v);
782 sqliteVdbeAddOp(v, OP_Ge, 0, addr+3, 0, 0);
783 sqliteVdbeAddOp(v, OP_Pop, 1, 0, 0, 0);
784 sqliteVdbeAddOp(v, OP_Goto, 0, dest, 0, 0);
785 sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
786 sqliteVdbeAddOp(v, OP_Gt, 0, dest, 0, 0);
787 break;
788 }
drhcce7d172000-05-31 15:34:51 +0000789 default: {
790 sqliteExprCode(pParse, pExpr);
791 sqliteVdbeAddOp(v, OP_Not, 0, 0, 0, 0);
792 sqliteVdbeAddOp(v, OP_If, 0, dest, 0, 0);
793 break;
794 }
795 }
796}
drh22827922000-06-06 17:27:05 +0000797
798/*
799** Do a deep comparison of two expression trees. Return TRUE (non-zero)
800** if they are identical and return FALSE if they differ in any way.
801*/
drhd8bc7082000-06-07 23:51:50 +0000802int sqliteExprCompare(Expr *pA, Expr *pB){
drh22827922000-06-06 17:27:05 +0000803 int i;
804 if( pA==0 ){
805 return pB==0;
806 }else if( pB==0 ){
807 return 0;
808 }
809 if( pA->op!=pB->op ) return 0;
drhd8bc7082000-06-07 23:51:50 +0000810 if( !sqliteExprCompare(pA->pLeft, pB->pLeft) ) return 0;
811 if( !sqliteExprCompare(pA->pRight, pB->pRight) ) return 0;
drh22827922000-06-06 17:27:05 +0000812 if( pA->pList ){
813 if( pB->pList==0 ) return 0;
814 if( pA->pList->nExpr!=pB->pList->nExpr ) return 0;
815 for(i=0; i<pA->pList->nExpr; i++){
drhd8bc7082000-06-07 23:51:50 +0000816 if( !sqliteExprCompare(pA->pList->a[i].pExpr, pB->pList->a[i].pExpr) ){
drh22827922000-06-06 17:27:05 +0000817 return 0;
818 }
819 }
820 }else if( pB->pList ){
821 return 0;
822 }
823 if( pA->pSelect || pB->pSelect ) return 0;
824 if( pA->token.z ){
825 if( pB->token.z==0 ) return 0;
826 if( pB->token.n!=pA->token.n ) return 0;
827 if( sqliteStrNICmp(pA->token.z, pB->token.z, pA->token.n)!=0 ) return 0;
828 }
829 return 1;
830}
831
832/*
833** Add a new element to the pParse->aAgg[] array and return its index.
834*/
835static int appendAggInfo(Parse *pParse){
836 if( (pParse->nAgg & 0x7)==0 ){
837 int amt = pParse->nAgg + 8;
838 pParse->aAgg = sqliteRealloc(pParse->aAgg, amt*sizeof(pParse->aAgg[0]));
839 if( pParse->aAgg==0 ){
840 sqliteSetString(&pParse->zErrMsg, "out of memory", 0);
841 pParse->nErr++;
842 return -1;
843 }
844 }
845 memset(&pParse->aAgg[pParse->nAgg], 0, sizeof(pParse->aAgg[0]));
846 return pParse->nAgg++;
847}
848
849/*
850** Analyze the given expression looking for aggregate functions and
851** for variables that need to be added to the pParse->aAgg[] array.
852** Make additional entries to the pParse->aAgg[] array as necessary.
853**
854** This routine should only be called after the expression has been
855** analyzed by sqliteExprResolveIds() and sqliteExprCheck().
856**
857** If errors are seen, leave an error message in zErrMsg and return
858** the number of errors.
859*/
860int sqliteExprAnalyzeAggregates(Parse *pParse, Expr *pExpr){
861 int i;
862 AggExpr *aAgg;
863 int nErr = 0;
864
865 if( pExpr==0 ) return 0;
866 switch( pExpr->op ){
867 case TK_FIELD: {
868 aAgg = pParse->aAgg;
869 for(i=0; i<pParse->nAgg; i++){
870 if( aAgg[i].isAgg ) continue;
871 if( aAgg[i].pExpr->iTable==pExpr->iTable
872 && aAgg[i].pExpr->iField==pExpr->iField ){
drh22827922000-06-06 17:27:05 +0000873 break;
874 }
875 }
876 if( i>=pParse->nAgg ){
877 i = appendAggInfo(pParse);
878 if( i<0 ) return 1;
879 pParse->aAgg[i].isAgg = 0;
880 pParse->aAgg[i].pExpr = pExpr;
881 }
drhaaf88722000-06-08 11:25:00 +0000882 pExpr->iAgg = i;
drh22827922000-06-06 17:27:05 +0000883 break;
884 }
885 case TK_AGG_FUNCTION: {
886 if( pExpr->iField==FN_Count || pExpr->iField==FN_Avg ){
887 if( pParse->iAggCount>=0 ){
888 i = pParse->iAggCount;
889 }else{
890 i = appendAggInfo(pParse);
891 if( i<0 ) return 1;
892 pParse->aAgg[i].isAgg = 1;
893 pParse->aAgg[i].pExpr = 0;
894 pParse->iAggCount = i;
895 }
896 if( pExpr->iField==FN_Count ){
897 pExpr->iAgg = i;
898 break;
899 }
900 }
901 aAgg = pParse->aAgg;
902 for(i=0; i<pParse->nAgg; i++){
903 if( !aAgg[i].isAgg ) continue;
drhd8bc7082000-06-07 23:51:50 +0000904 if( sqliteExprCompare(aAgg[i].pExpr, pExpr) ){
drh22827922000-06-06 17:27:05 +0000905 break;
906 }
907 }
908 if( i>=pParse->nAgg ){
909 i = appendAggInfo(pParse);
910 if( i<0 ) return 1;
911 pParse->aAgg[i].isAgg = 1;
912 pParse->aAgg[i].pExpr = pExpr;
913 }
914 pExpr->iAgg = i;
915 break;
916 }
917 default: {
918 if( pExpr->pLeft ){
919 nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pLeft);
920 }
921 if( nErr==0 && pExpr->pRight ){
922 nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pRight);
923 }
924 if( nErr==0 && pExpr->pList ){
925 int n = pExpr->pList->nExpr;
926 int i;
927 for(i=0; nErr==0 && i<n; i++){
928 nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pList->a[i].pExpr);
929 }
930 }
931 break;
932 }
933 }
934 return nErr;
935}